#!/usr/bin/env python3
# -*- coding:utf-8 -*-
# @Author: renjin@bit.edu.cn
# @Date  : 2024-08-08

from spirecv.dataloader.COCODatasetLoaderNode import COCODatasetLoaderNode
# from spirecv.algorithm.common_det.YOLOv8DetNode import YOLOv8DetNode
from spirecv.algorithm.common_det.YOLOv11MultiBatchDetJetsonTensorRTNode import YOLOv11MultiBatchDetJetsonTensorRTNode


job_name = 'detection_evaluation'
parameter_file = '../../params/spirecv2/default_params.json'
remote_ip = '59.110.144.11'


dataloader = COCODatasetLoaderNode(
    job_name, param_dict_or_file=parameter_file,
    remote_ip=remote_ip
)
algorithm = YOLOv11MultiBatchDetJetsonTensorRTNode(
    job_name, param_dict_or_file=parameter_file,
    specified_output_topic='/SpireView/Detection2DEval/TaskInput',
    remote_ip=remote_ip
)


dataloader.launch_next()
